"""沪深 A 股股票季度表现
# 默认为最新季度，亦可指定季度
"""
import efinance as ef
from typing import Literal
import os
current_path = os.getcwd()
print(current_path)
# 101: 日线 102: 周线 103: 月线 100: 指数 110: 经济数据
KLT = Literal[100, 101, 102, 103, 110]

data = ef.stock.get_all_company_performance() 

# 创建数据目录。
BASE = "."


# 存储相应股票代码数据到CSV
def get_filename(code: str, klt: KLT,stockName: str) -> str:
    """获取行情数据文件名。

    Args:
        code (str): 股票代码
        klt (KLT): 行情类型

    Returns:
        str: 行情数据文件名
    """
    return f"{BASE}/code/Stock/data/{klt}/{code}{stockName}.csv"
data.to_csv(get_filename('','110','沪深A股股票季度表现') , index=False)